heapdict==1.0.1
matplotlib==3.8.3
numpy==1.26.4
scipy==1.12.0
torch==2.1.1
